package com.vic.io.traverse;

import java.io.File;
import java.util.Arrays;
import java.util.LinkedList;

/**
 * 遍历文件
 * @author victor
 */
public class TraverseTest {

    public static void main(String[] args) {

        String outputDir = "/Users/renliwo/Downloads/update-pdf3";

        // 非递归方式
//        folderMethod1(outputDir);

        // 递归方式
        folderMethod2(outputDir);
    }


    public static void folderMethod1(String path) {
        int fileNum = 0;
        int folderNum = 0;
        File file = new File(path);
        LinkedList<File> list = new LinkedList<>();

        if (file.exists()) {
            if (null == file.listFiles()) {
                return;
            }
            list.addAll(Arrays.asList(file.listFiles()));
            while (!list.isEmpty()) {
                File curFile = list.removeFirst();

                // 文件
                if(curFile.isFile()) {
                    System.out.println("文件:" + curFile.getAbsolutePath());
                    fileNum++;
                    continue;
                }
                // 目录
                // 这里不要加进去，上面remove了，这里再加进去，下次遍历还有这个
//                list.add(curFile);
                folderNum++;

                File[] files = curFile.listFiles();
                if (null == files) {
                    continue;
                }
                for (File f : files) {
                    if (f.isDirectory()) {
                        System.out.println("文件夹:" + f.getAbsolutePath());
                        list.add(f);
                        folderNum++;
                    } else {
                        System.out.println("文件:" + f.getAbsolutePath());
                        fileNum++;
                    }
                }
            }
        } else {
            System.out.println("文件不存在!");
        }
        System.out.println("文件夹数量:" + folderNum + ",文件数量:" + fileNum);
    }

    public static void folderMethod2(String path) {
        File file = new File(path);
        if (file.exists()) {
            File[] files = file.listFiles();
            if (null != files) {
                for (File file2 : files) {
                    if (file2.isDirectory()) {
                        System.out.println("文件夹:" + file2.getAbsolutePath());
                        folderMethod2(file2.getAbsolutePath());
                    } else {
                        System.out.println("文件:" + file2.getAbsolutePath());
                    }
                }
            }
        } else {
            System.out.println("文件不存在!");
        }
    }
}
